package com.cg.leetcode;

import org.junit.Test;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * 144.二叉树的前序遍历
 *
 * @program: LeetCode->LeetCode_144
 * @author: cg
 * @create: 2022-04-18 20:47
 **/
public class LeetCode_144 {

    @Test
    public void test144() {
        TreeNode root = new TreeNode(1);
        root.right = new TreeNode(2);
        root.right.left = new TreeNode(3);
        System.out.println(preorderTraversal(root));
    }

    /**
     * 给你二叉树的根节点 root ，返回它节点值的 前序 遍历。
     * <p>
     * 示例 1：
     * <p>
     * 输入：root = [1,null,2,3]
     * 输出：[1,2,3]
     * <p>
     * 示例 2：
     * 输入：root = []
     * 输出：[]
     * <p>
     * 示例 3：
     * 输入：root = [1]
     * 输出：[1]
     * <p>
     * 示例 4：
     * 输入：root = [1,2]
     * 输出：[1,2]
     * <p>
     * 示例 5：
     * 输入：root = [1,null,2]
     * 输出：[1,2]
     * <p>
     * 提示：
     * 树中节点数目在范围 [0, 100] 内
     * -100 <= Node.val <= 100
     * <p>
     * 进阶：递归算法很简单，你可以通过迭代算法完成吗？
     *
     * @param root
     * @return
     */
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Deque<TreeNode> deque = new ArrayDeque<>();
        deque.push(root);
        while (!deque.isEmpty()) {
            TreeNode pop = deque.pop();
            res.add(pop.val);
            //前序遍历是中左右，每次先处理的是中间节点，那么先将根节点放入栈中，然后将右孩子加入栈，再加入左孩子。
            //为什么要先加入 右孩子，再加入左孩子呢？ 因为这样出栈的时候才是中左右的顺序。
            if (pop.right != null) {
                deque.push(pop.right);
            }
            if (pop.left != null) {
                deque.push(pop.left);
            }
        }
        return res;
    }

}
